package com.geeguo.ebuilder.core.security.utils;

import com.geeguo.ebuilder.core.security.config.LoginConfig;
import com.geeguo.ebuilder.core.security.constants.SecurityConstants;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.security.Keys;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.crypto.SecretKey;
import java.nio.charset.StandardCharsets;
import java.util.Map;

/**
 * Jwt工具类
 */
@Component
public class JwtUtils implements InitializingBean {

    private static LoginConfig loginConfig;
    private static SecretKey   key;

    @Autowired
    public void setLoginConfig(LoginConfig loginConfig) {
        JwtUtils.loginConfig = loginConfig;
    }

    @Override
    public void afterPropertiesSet() throws Exception {
        key = Keys.hmacShaKeyFor(loginConfig.getTokenSecretKey().getBytes(StandardCharsets.UTF_8));
    }

    /**
     * 使用数据声明生成令牌
     *
     * @param claims 数据声明
     * @return 令牌
     */
    public static String createToken(Map<String, String> claims) {
        return Jwts.builder()
                // 设置JWT的声明部分
                .claims(claims)
                // 对JWT进行签名
                .signWith(key, Jwts.SIG.HS256)
                .compact();
    }

    /**
     * 从令牌中获取数据声明
     *
     * @param token 令牌
     * @return 数据声明
     */
    public static Claims parseToken(String token) {
        return Jwts.parser()
                .verifyWith(key)
                .build()
                .parseSignedClaims(token)
                .getPayload();
    }

    /**
     * 从数据声明中获取信息
     *
     * @param claims 声明信息
     * @param key 键
     * @return 值
     */
    public static String getValue(Claims claims, String key) {
        if (claims != null && claims.get(key) != null) {
            return claims.get(key).toString();
        }
        return null;
    }

    /**
     * 获取用户登录标识
     * 
     * @param token 令牌
     * @return 用户登录标识
     */
    public static String getLoginKey(String token) {
        Claims claims = parseToken(token);
        return getValue(claims, SecurityConstants.LOGIN_KEY);
    }

    /**
     * 获取用户登录标识
     * 
     * @param claims 声明信息
     * @return 用户登录标识
     */
    public static String getLoginKey(Claims claims) {
        return getValue(claims, SecurityConstants.LOGIN_KEY);
    }

    /**
     * 获取用户ID
     * 
     * @param token 令牌
     * @return 用户ID
     */
    public static String getUserId(String token) {
        Claims claims = parseToken(token);
        return getValue(claims, SecurityConstants.USER_ID);
    }

    /**
     * 获取用户ID
     * 
     * @param claims 声明信息
     * @return 用户ID
     */
    public static String getUserId(Claims claims) {
        return getValue(claims, SecurityConstants.USER_ID);
    }
}